from flask_restful import Resource, reqparse, request
from comment.models.index import *
from comment.utils.data2dict import datalist2dict, data2dict
from comment.utils.shopping_redis import redis_client
import json


# 商品分类的资源类
class Shopping_Category(Resource):

    def get(self):
        # 从redis读取数据
        data_cache = redis_client.get("index_category")
        if data_cache: # 如果redis数据中有数据就返回数据
            return json.loads(data_cache)
        else: # 如果没有数据则从MySQL查询数据返回给用户写入redis
            rq = reqparse.RequestParser()
            rq.add_argument('parent_id', required=True, type=int)
            req = rq.parse_args()
            # 资源类 一级分类
            data = self.getData(req.parent_id)
            if data:
                for item in data:
                    item.update({'list': ''})
                    data_second = self.getData(item["id"])
                    item['list'] = data_second
                    for item1 in data_second:
                        item.update({'list': ''})
                        data_third = self.getData(item1["id"])
                        item1['list'] = data_third
                # 把查询到的数据写入redis中   # 缓存的过期时间
                redis_client.setex('index_category', 3600*24, json.dumps(data))
                return data
            else:
                return {'message': 'none'}


    @staticmethod
    def getData(parent_id):
        res = Category.query.with_entities(Category.id, Category.parent_id, Category.name) \
            .filter(Category.parent_id == parent_id).order_by(Category.sort.asc()).all()
        if res:
            return datalist2dict(res)
        else:
            return 'none'

class Dict_test(Resource):
    def get(self):
        # 取多条数据
        res = Category.query.with_entities(Category.id,Category.name).filter(Category.parent_id==0).limit(10).all()

        if res:
            return datalist2dict(res)
        else:
            return 'none'

# 首页新品推荐的资源类
class Shopping_HomeNewProduct(Resource):

    def get(self):
        res = HomeNewProduct.query.join(Product,HomeNewProduct.product_id == Product.id)\
            .with_entities(Product.id,Product.default_pic,Product.price,Product.product_name,
                           Product.rel_category3_id).order_by(HomeNewProduct.sort.asc())\
            .limit(10).all()
        if res:
            return datalist2dict(res)
        else:
            return {'message':'none'}

# 首页人气热搜的资源类
class Shopping_HomeRecommendProduct(Resource):

    def get(self):
        data_cache = redis_client.get('index_HomeRecommendProduct')
        if data_cache:
            return json.loads(data_cache)
        else:
            res = HomeRecommendProduct.query.join(Product,HomeRecommendProduct.product_id==Product.id)\
                .with_entities(Product.id,Product.default_pic,Product.product_name,Product.rel_category3_id,Product.price)\
                .limit(10).all()
            if res:
                data = datalist2dict(res)
                # 将查询数据写入到redis数据库中
                redis_client.setex('index_HomeRecommendProduct',3600*24,json.dumps(data))
                return data
            else:
                return {'message':'none'}

# 首页专题的资源类 # 专题广场
class Shopping_RecommendSubject(Resource):

    def get(self):
        data_cache = redis_client.get('index_RecommendSubject')

        # 从redis中读取数据
        if data_cache:
            return json.loads(data_cache)
        else:
            res = CmsSubject.query.filter(CmsSubject.show_status==1).all()
            if res:
                data = datalist2dict(res)
                for i in range(len(data)):
                    res_product = CmsSubjectProductRelation.query.join(Product,CmsSubjectProductRelation.product_id==Product.id)\
                    .filter(CmsSubjectProductRelation.subject_id==data[i]["id"]).with_entities\
                        (Product.id,Product.default_pic,Product.price,Product.product_name,Product.rel_category3_id)\
                    .limit(10).all()
                    if res_product:
                        data[i]["productList"] = datalist2dict(res_product)
                    else:
                        data[i]["productList"] = ''
                # 将查询到的数据写入redis
                redis_client.setex('index_RecommendSubject',3600*24,json.dumps(data))
                return data
            else:
                return {'message':'none'}

